#!/bin/bash
ID=$1
DATASET=$2
SHIFT=$3
SUFFIX=$4
D_lr=$5
CLIP=$6
OUT=../data/results/${DATASET}_${SHIFT}_${SUFFIX}
if [ ! -d $OUT ]; then
  mkdir $OUT
fi
# $ ./cmd/train.sh 0 tvsum_aug sup hs2048_2018 1e-6
# echo ${ID}_${DATASET}_${SHIFT}_${SUFFIX}
for ((i=1; i<=5; i++))
do
    CUDA_VISIBLE_DEVICES=$ID python train.py \
        --dataset $DATASET \
        --shift $SHIFT \
        --fold $i \
        --suffix ${SUFFIX} \
        --n_epochs 30 \
        --discriminator_slow_start 15 \
        --discriminator_lr $D_lr \
        --clip $CLIP\
        > ${OUT}/${DATASET}_${SHIFT}_${i}_${SUFFIX}.log 2>&1 &
    wait
done
